Skip to content

ENH: Add Model class for model training and validation #19

Merged
merged 3 commits into from
Sep 24, 2023
Merged

Conversation

deshmukg
Copy link
Contributor

@deshmukg deshmukg commented Sep 24, 2023

  • Added Model wrapper class around the MultiGCN model for training, validation, and prediction.
  • Disabled batch_norm by default in CGConv.
  • Created a util function create_dataloaders to create training, validation, and test dataloaders.

@deshmukg deshmukg merged commit b6b8292 into master Sep 24, 2023
Sign in to join this conversation on GitHub.
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

1 participant